from tensorflow.keras import Model


class ModelTrainner:
    """
    模型训练器
    """

    def __init__(self, epochs: int, batch_size: int):
        self.epochs = epochs
        self.batch_size = batch_size

    def train(self) -> None:
        print("trainning--------\n epochs:", self.epochs, "  batch_size", self.batch_size)
        return
